Memory Decoder at Scale: A Pretrained, Parametric Long-Term Memory
大规模参数化长期记忆模块:独立预训练与解耦扩展
🧭 参数化长期记忆与 LLM 解耦 · 独立扩展 ★ 深度解读
LUMIA Lab, Shanghai Jiao Tong University · Shanghai Artificial Intelligence Laboratory · Tsinghua University
将参数化长期记忆模块扩展至 6.9B 参数、300B 令牌预训练,小 backbone 搭配大记忆在多项基准上超越更大纯骨干模型且总参数更少
📌 工作参考价值(4 点)
- 记忆模块即插即用,适合多领域推荐系统快速迁移:预训练好的通用或领域记忆可以直接插在冻结的推荐模型(如同 backbone)上,仅需少量计算即适配新领域,避免全量微调,显著降低部署成本。
- kNN 分布蒸馏到参数模块,避免在线检索:电商搜索或推荐场景通常需要实时反馈,该方案将检索信号离线蒸馏为参数化记忆,推理时无需外部索引和检索,降低延迟和存储,可直接用于排序或召回的特征增强。
- 小 backbone + 大记忆的参数效率优势:在推荐模型中,可以用相对较小的主网络搭配较大记忆模块,在总参数受限时比单纯扩大主网络更高效,例如为轻量召回模型配备大容量记忆来提升冷门物品或长尾查询的覆盖。
- 记忆训练目标可提升训练数据可追溯性:论文中的抽取式记忆评估表明记忆能更精准还原训练样本,类似思路可用于推荐系统中关键行为或商品的记忆固化,帮助可解释性和合规审查。